package com.itheima.servlet;

import com.itheima.utils.CookieUtil;

import javax.servlet.annotation.WebServlet;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;


@WebServlet("/voteServlet")
public class VoteServlet extends HttpServlet {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
        //通过工具类拿cookie   这个cookie记录投票的次数
        Cookie[] cookies = req.getCookies();
        Cookie cookieVoteNumber = CookieUtil.getCookieByName(cookies,"voteNumber");
        //投票的次数
        if(cookieVoteNumber==null){
            //第一次提交没有cookie
            cookieVoteNumber = new Cookie("voteNumber", "1");
            //添加
            resp.addCookie(cookieVoteNumber);
            //在这里写入数据库加一


            //重定向
            resp.sendRedirect("vote.jsp");
            return;
        }
        //不是第一次投判断投的次数  这个是记录的投票次数
        String voteNumber = cookieVoteNumber.getValue();
        int number = Integer.parseInt(voteNumber);
        if(number==10){
            //已经投了3次
            Cookie cookie = new Cookie("message", "投票次数上限");
            resp.addCookie(cookie);
            resp.sendRedirect("vote.jsp");
            return;
        }
        //在这里向数据库写入票数加一


        //没有到三次  次数加一
        number = number + 1;
        //设置cookie新的值
        cookieVoteNumber.setValue("" + number);
        resp.addCookie(cookieVoteNumber);
        //重定向
        resp.sendRedirect("vote.jsp");
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws  IOException {
        this.doGet(req, resp);
    }
}